﻿#region Copyright

// The contents of this file are subject to the Mozilla Public License
//  Version 1.1 (the "License"); you may not use this file except in compliance
//  with the License. You may obtain a copy of the License at
//  
//  http://www.mozilla.org/MPL/
//  
//  Software distributed under the License is distributed on an "AS IS"
//  basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
//  License for the specific language governing rights and limitations under 
//  the License.
//  
//  The Initial Developer of the Original Code is Robert Smyth.
//  Portions created by Robert Smyth are Copyright (C) 2008.
//  
//  All Rights Reserved.

#endregion

using System;
using System.IO;
using System.IO.Pipes;
using NUnitGrid.Framework;


namespace NUnitGrid.Server.Plugins.GridNodesLink
{
    public class GridNodeLink : IGridNodesLink, IScheduledEventListener
    {
        private const int maxNumberOfConnections = 100;
        private readonly IGridNodeLinkListener listener;
        private readonly ISchedule schedule;
        private readonly TimeSpan tickPeriod = TimeSpan.FromMilliseconds(25);
        private bool connectionMade;
        private NamedPipeServerStream pipeServer;

        public GridNodeLink(ISchedule schedule, IGridNodeLinkListener listener)
        {
            this.schedule = schedule;
            this.listener = listener;
        }

        public void Start()
        {
            pipeServer = new NamedPipeServerStream("NUnitGrid", PipeDirection.InOut, maxNumberOfConnections,
                                                   PipeTransmissionMode.Message,
                                                   PipeOptions.Asynchronous);
            schedule.AddEvent(this, tickPeriod);
        }

        public void ListenForConnection()
        {
            pipeServer.BeginWaitForConnection(OnConnect, pipeServer);
        }

        public void SendCommand(string message)
        {
            using (var streamWriter = new StreamWriter(pipeServer))
            {
                streamWriter.AutoFlush = true;
                streamWriter.WriteLine(Console.ReadLine());
            }
        }

        void IScheduledEventListener.OnEventFired()
        {
            if (connectionMade)
            {
                connectionMade = false;
                listener.OnConnect(true);
            }

            schedule.AddEvent(this, tickPeriod);
        }

        private void OnConnect(IAsyncResult result)
        {
            try
            {
                var pipServerStream = (NamedPipeServerStream) result.AsyncState;
                pipServerStream.EndWaitForConnection(result);
                connectionMade = true;
            }
            catch (Exception)
            {
            }
        }
    }
}